working with the pytorch autograd framework